import gymnasium as gym
import highway_env

from matplotlib import pyplot as plt

env = gym.make(
  "highway-v0",
  config={"lanes_count": 2},
  render_mode='rgb_array',
)

env.reset()

for _ in range(300):
    action = env.unwrapped.action_type.actions_indexes['IDLE']
    obs, reward, done, truncated, info = env.step(action)
    env.render()

plt.imshow(env.render())
plt.show()
